7bbde3eb03deaec82bc311a24e514f43e2eb6693,src/main/java/hex/nn/NN.java,NN,buildModel,#NNModel#,253

Before Change


    Log.info("Memory usage of the model: " + String.format("%.2f", (double)model_size*Float.SIZE / (1<<23)) + " MB.");

    final Frame train = model.model_info().data_info()._adaptedFrame;
    Frame trainScoreFrame = sampleFrame(train, score_training_samples, seed);

    Frame[] valid_adapted = null;
    Frame valid = null;

After Change


      Log.info("Number of model parameters (weights/biases): " + String.format("%,d", model_size));
      Log.info("Memory usage of the model: " + String.format("%.2f", (double)model_size*Float.SIZE / (1<<23)) + " MB.");
      train = reBalance(model.model_info().data_info()._adaptedFrame, seed);
      trainScoreFrame = sampleFrame(train, score_training_samples, seed);
      Log.info("Number of chunks of the training data: " + train.anyVec().nChunks());
      if (validation != null) {
        valid_adapted = model.adapt(validation, false);